#!/usr/bin/python
#coding=utf-8
'''
Crea l'albero contenente i dati per lo studio dei conteggi di luce e buio a
diversi voltaggi del PMT
---
---
Crea un albero contenente tre rami: volt, luce, buio
'''

from __future__ import division, print_function
from ROOT import TTree, TFile, AddressOf
from array import array
from myanalisis import my_double

'''
Definizione dei parametri e inizializzazione degli oggetti
'''
ofile_name = 'volt_luce_buio.root'
ofile = TFile(ofile_name, 'recreate')
tree = TTree('volt_luce_buio', 'Conteggi di luce e buio per diversi voltaggi del PMT')
volt = my_double()
luce = my_double()
buio = my_double()
tree.Branch('volt', AddressOf(volt, 'value'), 'value/D')
tree.Branch('luce', AddressOf(luce, 'value'), 'value/D')
tree.Branch('buio', AddressOf(buio, 'value'), 'value/D')

volts = [700, 650, 600, 544, 501, 450, 402]
time_luce = [10, 50, 100, 140, 180, 230, 270]
time_buio = [30, 70, 120, 160, 200, 250, 290]

'''
Estrae i valori dal file nella cartella raw/ e li salva nell'albero
'''
for i, volt.value in enumerate(volts):
    with open('raw/Misure_Volt-TimeAcq') as ifile:
        for line in ifile:
            if line.split()[0] == str(time_luce[i]):
                luce.value = float(line.split()[9])
            if line.split()[0] == str(time_buio[i]):
                buio.value = float(line.split()[9])

        tree.Fill()

ofile.Write()
ofile.Close()
